#include "htf_ai_interface.h"
#include "../utils/htf_logger.h"

int main(int argc, char* argv[]) {
    if (argc != 2) {
        HTF_LOG_ERROR("Usage: %s <model_path>", argv[0]);
        return 1;
    }

    htf_ai_handle_t* ai = htf_ai_create(argv[1]);
    if (!ai) {
        HTF_LOG_ERROR("Failed to create AI handle");
        return 1;
    }

    float input[10] = {1,2,3,4,5,6,7,8,9,10};
    float output[5] = {0};

    if (htf_ai_infer(ai, input, 10, output, 5) != 0) {
        HTF_LOG_ERROR("Inference failed");
        htf_ai_destroy(ai);
        return 1;
    }

    HTF_LOG_INFO("Inference result: %f %f %f %f %f",
                 output[0], output[1], output[2], output[3], output[4]);

    htf_ai_destroy(ai);
    return 0;
}